/*
 *  Copyright (C) [2024] smartboot [zhengjunweimail@163.com]
 *
 *  企业用户未经smartboot组织特别许可，需遵循AGPL-3.0开源协议合理合法使用本项目。
 *
 *   Enterprise users are required to use this project reasonably
 *   and legally in accordance with the AGPL-3.0 open source agreement
 *  without special permission from the smartboot organization.
 */

package tech.smartboot.feat.ai.reranker;

import tech.smartboot.feat.ai.Options;

/**
 * @author 三刀 zhengjunweimail@163.com
 * @version v1.0.0
 */
public class RerankerOptions extends Options {

    private RerankerModelVendor model;

    public RerankerOptions debug(boolean debug) {
        super.debug(debug);
        return this;
    }


    public RerankerOptions baseUrl(String baseUrl) {
        super.baseUrl(baseUrl);
        return this;
    }


    public RerankerOptions model(RerankerModelVendor model) {
        this.model = model;
        return baseUrl(model.baseUrl());
    }

    public RerankerModelVendor model() {
        return model;
    }

    public RerankerOptions apiKey(String apiKey) {
        super.apiKey(apiKey);
        return this;
    }
}
